from langchain_core.prompts import ChatPromptTemplate
from tools import get_llm

messages = ChatPromptTemplate(
    [
        ('system', '你是一个翻译官，你需要把用户的语言翻译成{language}'),
        ('human', '{content}')
    ]
)

# 提示词模板处理
question = {
    'language': '英语',
    'content': '今天天气真好'
}

# 调用message中的invoke
msg = messages.invoke({'language': '英语', 'content': '今天天气真好'})
llm = get_llm()
# 调用message中的format
msg1 = messages.format(language='英文', content='今天天气真好')
result = llm.invoke(msg)

result1 = llm.invoke(msg1)
print(result)
print(result1)
